tensor([[0.3700, 0.3700, 0.3700, 0.3700, 0.3700],
        [0.6300, 0.6300, 0.6300, 0.6300, 0.6300]], grad_fn=<RoundBackward1>)